import torch
import numpy as np
from model import Transformer